home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / gufw / model / Firewall.py < prev    next >
Encoding:
Python Source  |  2009-10-25  |  5.7 KB  |  180 lines
  1. # Gufw 9.10.4 - http://gufw.tuxfamily.org
  2. # Copyright (C) 2009 Marcos Alvarez Costales
  3. #
  4. # Gufw is free software; you can redistribute it and/or modify
  5. # it under the terms of the GNU General Public License as published by
  6. # the Free Software Foundation; either version 3 of the License, or
  7. # (at your option) any later version.
  8. # 
  9. # Gufw is distributed in the hope that it will be useful,
  10. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12. # GNU General Public License for more details.
  13. # 
  14. # You should have received a copy of the GNU General Public License
  15. # along with Gufw; if not, see http://www.gnu.org/licenses for more
  16. # information.
  17.  
  18.  
  19. from FrontEnd import Frontend
  20. from Log      import Log
  21. from Variable import Variable
  22.  
  23.  
  24. class Firewall:
  25.  
  26.     def __init__(self):
  27.     
  28.         self.rules_list = []
  29.         
  30.         self.frontend = Frontend()
  31.         self.log      = Log()
  32.         self.variable = Variable()
  33.         
  34.         self.status    = self.frontend.get_status()
  35.         self.default   = self.frontend.get_default()
  36.         self.ufw_log   = self.frontend.get_ufw_log()
  37.         self.gufw_log  = self.frontend.get_gufw_log()
  38.         self.wrap      = self.log.get_wrapping()
  39.         self.maximized = False
  40.  
  41.  
  42.     # Return status ufw
  43.     def set_status(self, status):
  44.         self.status = self.frontend.set_status(status)
  45.         self.log.add_log(self.gufw_log, self.variable.get_command(status))
  46.         
  47.  
  48.     # Get status ufw
  49.     def get_status(self):
  50.         return self.status
  51.  
  52.  
  53.     # Return status ufw
  54.     def set_default(self, default):
  55.         self.default = self.frontend.set_default(default)
  56.         self.log.add_log(self.gufw_log, self.variable.get_command(default))
  57.  
  58.  
  59.     # Get status ufw 
  60.     def get_default(self):
  61.         return self.default
  62.         
  63.  
  64.     # Get actual rules split by Action
  65.     def get_rules_list(self):
  66.         self.rules_list = []
  67.         
  68.         rules = self.frontend.get_rules()
  69.         
  70.         for rule in rules:
  71.             self.rules_list.append(self.get_rule_split_by_action(rule))
  72.             
  73.         return self.rules_list
  74.  
  75.  
  76.     # Refresh Log
  77.     def refresh_log(self):
  78.         self.log.refresh_log()
  79.  
  80.  
  81.     # Get Log
  82.     def get_log(self):
  83.         return self.log.get_log()
  84.         
  85.  
  86.     # Get Wrapping
  87.     def get_wrap(self):
  88.         return self.log.get_wrapping()
  89.         
  90.         
  91.     # Set Wrapping
  92.     def set_wrap(self, wrapping):
  93.         self.wrap = self.log.set_wrapping(wrapping)
  94.  
  95.     
  96.     # Get ufw log
  97.     def get_ufw_log(self):
  98.         return self.ufw_log
  99.         
  100.         
  101.     # Set ufw log
  102.     def set_ufw_log(self, status):
  103.         self.ufw_log = self.frontend.set_ufw_log(status)
  104.         self.log.add_log(self.gufw_log, self.variable.get_command(status))
  105.     
  106.     
  107.     # Get Gufw log
  108.     def get_gufw_log(self):
  109.         return self.gufw_log
  110.         
  111.         
  112.     # Set Gufw log
  113.     def set_gufw_log(self, status):
  114.         self.gufw_log = self.frontend.set_gufw_log(status)
  115.         if self.gufw_log == self.variable.get_constant("gufw_log_on"):
  116.             self.log.add_log(self.gufw_log, self.variable.get_text("004"))
  117.         else:
  118.             self.log.add_log("gufw_log_on", self.variable.get_text("005"))
  119.     
  120.      
  121.     # Add rule to ufw
  122.     def add_rule(self, service, action, protocol, fromip, fromport, toip, toport):
  123.         rule = self.frontend.add_rule_component(service, action, protocol, fromip, fromport, toip, toport)
  124.         self.log.add_log(self.gufw_log, rule)
  125.         return self.frontend.add_rule(rule)
  126.         
  127.  
  128.     # Remove rule from ufw
  129.     def remove_rule(self, rule_to, action, rule_from):
  130.         rule = self.frontend.remove_rule_component(rule_to, action, rule_from)
  131.         self.log.add_log(self.gufw_log, rule)
  132.         return self.frontend.remove_rule(rule)
  133.  
  134.  
  135.     # Split rule by action
  136.     def get_rule_split_by_action(self, rule):
  137.  
  138.         return_split_rule = []
  139.         
  140.         # Deny?
  141.         if rule.find(self.variable.get_constant("deny_upper")) != -1:
  142.             split_rule = rule.split(self.variable.get_constant("deny_upper"))
  143.             return_split_rule.append(split_rule[0].strip())
  144.             return_split_rule.append(self.variable.get_constant("deny_upper"))
  145.             return_split_rule.append(split_rule[1].strip())
  146.  
  147.         # Allow?
  148.         elif rule.find(self.variable.get_constant("allow_upper")) != -1:
  149.             split_rule = rule.split(self.variable.get_constant("allow_upper"))
  150.             return_split_rule.append(split_rule[0].strip())
  151.             return_split_rule.append(self.variable.get_constant("allow_upper"))
  152.             return_split_rule.append(split_rule[1].strip())
  153.             
  154.         # Limit?
  155.         elif rule.find(self.variable.get_constant("limit_upper")) != -1:
  156.             split_rule = rule.split(self.variable.get_constant("limit_upper"))
  157.             return_split_rule.append(split_rule[0].strip())
  158.             return_split_rule.append(self.variable.get_constant("limit_upper"))
  159.             return_split_rule.append(split_rule[1].strip())
  160.             
  161.         # Reject?
  162.         elif rule.find(self.variable.get_constant("reject_upper")) != -1:
  163.             split_rule = rule.split(self.variable.get_constant("reject_upper"))
  164.             return_split_rule.append(split_rule[0].strip())
  165.             return_split_rule.append(self.variable.get_constant("reject_upper"))
  166.             return_split_rule.append(split_rule[1].strip())
  167.             
  168.         return return_split_rule
  169.  
  170.  
  171.     # Size Gufw window
  172.     def get_old_size_window(self):
  173.         width,height = self.frontend.get_old_size_window()
  174.         return width,height
  175.         
  176.         
  177.     # Save actual size window
  178.     def save_size_window(self, win_width, win_height):
  179.         self.frontend.save_size_window(win_width, win_height)
  180.